from models.experimental import attempt_load
import torch
import pandas as pd
from utils.general import strip_optimizer

#转换模型的name
model_path = r"/Users/edy/Downloads/YQSL_CPT_STATIC_20211126.pt"
model_path = r"/Users/edy/Downloads/YQSL_DP_STATIC_20211230.pt"
model_path = r'/Users/edy/Downloads/get_drink/YQSL_CPT_STATIC_20211126_sku.pt'
output_model_path = r'/Users/edy/Downloads/get_drink/YQSL_DP_STATIC_20211230_sku.pt'
model = attempt_load(model_path, map_location='cpu')
replace_names = []
csv_path = r"/Users/edy/Downloads/元气森林-贩冰冰-商品识别编码库 - 汇总.csv"

data = pd.read_csv(csv_path)
sku_code = data['sku_code']
sku_name = data['sku_name']
name_to_code_dict = {}
for idx in range(len(sku_code)):
    name_to_code_dict[sku_name[idx]] = sku_code[idx]

# replace_names=["DPS002F002S01T1W6P2250 东鹏特饮维生素功能饮料罐装250毫升",
# "DPS056F090S01T1W30P1400 东鹏陈皮特饮瓶装400毫升",
# "DPS332F001S01T1W24P2355 东鹏加気能量型营养素饮料罐装355毫升",
# "DPS056F090S01T1W29P9310 东鹏由柑柠檬茶高罐装310毫升",
# "DPS357F057S03T1W9P1380 饮用天然水",
# "DPS002F002S01T1W6P1250 东鹏特饮维生素功能饮料瓶装250毫升",
# "DPS002F001S03T1W6P2335 东鹏0糖无糖型能量饮料",
# "DPS356F003S02T1W20P1330 东鹏大咖®咖啡饮料",
# "DPS356F003S02T1W20P1500 东鹏大咖®咖啡饮料",
# "DPS002F002S01T1W6P1500 东鹏特饮维生素功能饮料瓶装500毫升",
# 'K0ANVCF00010001000 Never Coffee-咖啡饮料-拿铁咖啡-270ml',
# 'K0ANVCF00020001000 Never Coffee-咖啡饮料-燕麦咖啡-270ml',
# 'K0ANVCF00030001000 Never Coffee-咖啡饮料-美式咖啡-270ml',
# 'K1AWXR00010001000 外星人-电解质水-荔枝海盐-电解质饮料-PET瓶-500ml',
# 'K1AWXR00020001000 外星人-电解质水-青柠-电解质饮料-PET瓶-500ml',
# 'K1AWXR00030001000 外星人-电解质水-西柚-电解质饮料-PET瓶-500ml',
# 'K1AWXR00040001000 外星人-能量饮料-经典-能量饮料-PET瓶-350ml',
# 'K1AWXR00040003000 外星人-能量饮料-经典-能量饮料-铝罐-330ml',
# 'K1AWXR00050001000 外星人-能量饮料-马黛茶-能量饮料-PET瓶-350ml',
# 'K1AWXR00050003000 外星人-能量饮料-马黛茶-能量饮料-铝罐-330ml',
# 'K1AWXR00070001000 外星人-能量饮料-乳酸菌-能量饮料-PET瓶-350ml',
# 'K1AWXR00070003000 外星人-能量饮料-乳酸菌-能量饮料-铝罐-330ml',
# 'K1AWXR00080001000 外星人-能量饮料-西柚-能量饮料-PET瓶-350ml',
# 'K1AWXR00080003000 外星人-能量饮料-西柚-能量饮料-铝罐-330ml',
# 'K1AWXR00090001000 外星人-能量饮料-彩虹独角兽-PET瓶-350ml',
# 'K1AYQSL00180001000 元气森林-元气森林-白桃-苏打气泡水-PET瓶-1250ml -福気',
# 'K1AYQSL00180003000 元气森林-元气森林-白桃-苏打气泡水-PET瓶-480ml',
# 'K1AYQSL00180007000 元气森林-元气森林-白桃-苏打气泡水-铝罐-200ml',
# 'K1AYQSL00190001000 元气森林-元气森林-卡曼橘-苏打气泡水-PET瓶-480ml',
# 'K1AYQSL00190005000 元气森林-元气森林-卡曼橘-苏打气泡水-铝罐-200ml',
# 'K1AYQSL00210001000 元气森林-元气森林-荔枝-苏打气泡水-PET瓶-480ml',
# 'K1AYQSL00220001000 元气森林-元气森林-海盐菠萝-苏打气泡水-PET-480ml',
# 'K1AYQSL00230001000 元气森林-元气森林-青瓜-苏打气泡水-PET瓶-480ml',
# 'K1AYQSL00240002000 元气森林-元气森林-乳酸菌-苏打气泡水-PET瓶-480ml',
# 'K1AYQSL00250002000 元气森林-元气森林-酸梅汁-苏打气泡水-PET瓶-480ml',
# 'K1AYQSL00260001000 元气森林-元气森林-樱花白葡萄-苏打气泡水-PET瓶-480ml',
# 'K1AYQSL00290001000 元气森林-燃茶-草莓茉莉-无糖乌龙茶-PET瓶-500ml',
# 'K1AYQSL00300001000 元气森林-燃茶-醇香-无糖乌龙茶-PET瓶-500ml',
# 'K1AYQSL00310001000 元气森林-燃茶-桃香-无糖乌龙茶-PET瓶-500ml',
# 'K1AYQSL00320001000 元气森林-燃茶-玄米-无糖乌龙茶-PET瓶-500ml',
# 'K1AYQSL00340002000 元气森林-元气森林乳茶-咖啡拿铁-乳茶-PET瓶-450ml',
# 'K1AYQSL00350002000 元气森林-元气森林乳茶-茉香奶绿-乳茶-PET瓶-450ml',
# 'K1AYQSL00360002000 元气森林-元气森林乳茶-浓香原味-乳茶-PET瓶-450ml',
# 'K1AYQSL00410001000 元气森林-元气森林-夏黑葡萄-苏打气泡水-PET瓶-480ml',
# 'K1AYQSL00420001000 元气森林-满分-西柚-果汁微气泡-PET瓶-380ml',
# 'K1AYQSL00430001000 元气森林-满分-夏黑葡萄-果汁微气泡-PET瓶-380ml',
# 'K1AYQSL00440001000 元气森林-满分-王林青苹果-果汁微气泡-PET瓶-380ml',
# 'K1AYQSL00450001000 元气森林-元气森林-王林青苹果-苏打气泡水-PET瓶-480ml',
# 'K1AYQSL00490001000 元气森林-燃茶-焙火-无糖乌龙茶-PET瓶-500ml',
# 'K1AYQSL00670001000 元气森林-元气森林-百香果-苏打气泡水-PET-480ml',
# 'K1AYQSL00680001000 元气森林-元气森林-石榴红树莓-苏打气泡水-PET-480ml',
# 'K1AYQSL00690001000 元气森林-纤茶-玉米须茶-PET-500ml',
# 'K1AYQSL00710001000 元气森林-元气森林-红香酥梨-苏打气泡水-PET-480ml',
# 'bottle',
# 'row',
# 'LHS061F002S01T1W6P1380 乐虎氨基酸维生素功能饮料瓶装380毫升',
# 'JLBS086F114S01T1W6P1560 健力宝运动饮料橙蜜味瓶装560毫升',
# 'NFSS025F001S03T1W17P1500 农夫山泉东方树叶茉莉花茶瓶装500毫升',
# 'NFSS029F058S01T1W6P1550 农夫山泉尖叫多肽型瓶装550毫升',
# 'NFSS029F005S01T1W6P1550 农夫山泉尖叫纤维型瓶装550毫升',
# 'NFSS030F036S01T1W18P1445 农夫山泉水溶C100柠檬味复合果汁饮料瓶装445毫升',
# 'NFSS030F240S01T1W7P1445 农夫山泉水溶C100西柚味复合果汁饮料瓶装445毫升',
# 'NFSS032F064S01T1W24P1500 农夫山泉维他命水柑橘风味瓶装500毫升',
# 'NFSS026F049S01T1W16P1500 农夫山泉茶π柚子绿茶瓶装500毫升',
# 'NFSS026F036S01T1W15P1500 农夫山泉茶π柠檬红茶瓶装500毫升',
# 'NFSS026F051S01T1W10P1500 农夫山泉茶π蜜桃乌龙茶瓶装500毫升',
# 'NFSS026F009S01T1W17P1500 农夫山泉茶π西柚茉莉花茶瓶装500毫升',
# 'NFSS024F057S03T1W9P1550 农夫山泉饮用天然水瓶装550毫升',
# 'JDBS096F001S01T1W30P1550 加多宝凉茶植物饮料瓶装550毫升',
# 'JDBS096F001S01T1W30P2310 加多宝凉茶植物饮料罐装310毫升',
# 'CCS003F001S01T1W2P1300 可口可乐瓶装300毫升',
# 'CCS003F001S01T1W2P1600 可口可乐瓶装600毫升',
# 'CCS003F001S01T1W2P2330 可口可乐罐装330毫升',
# 'CCS005F001S01T1W2P1500 可口可乐零度瓶装500毫升',
# 'HQZS108F001S01T1W30P1550 和其正凉茶瓶装550毫升',
# 'WHHS342F125S01T1W32P1450 娃哈哈AD钙奶瓶装450毫升',
# 'WHHS329F125S01T1W32P1500 娃哈哈营养快线果汁牛奶瓶装500毫升',
# 'WHHS133F001S03T1W9P1596 娃哈哈饮用纯净水瓶装596毫升',
# 'MKS138F100S01T1W7P11000 康师傅冰糖雪梨瓶装1000毫升',
# 'MKS138F100S01T1W7P1500 康师傅冰糖雪梨瓶装500毫升',
# 'MKS161F036S01T1W15P11000 康师傅冰红茶柠檬口味瓶装1000毫升',
# 'MKS161F036S01T1W15P1500 康师傅冰红茶柠檬口味瓶装500毫升',
# 'MKS166F001S03T1W9P1550 康师傅喝开水熟水饮用水塑料瓶550毫升',
# 'MKS164F001S02T1W16P1500 康师傅绿茶低糖瓶装500毫升',
# 'MKS164F001S01T1W16P11000 康师傅绿茶瓶装1000毫升',
# 'MKS164F147S01T1W30P11000 康师傅茉莉清茶瓶装1000毫升',
# 'MKS164F147S01T1W30P1500 康师傅茉莉清茶瓶装500毫升',
# 'MKS164F148S01T1W30P11000 康师傅茉莉蜜茶瓶装1000毫升',
# 'MKS164F148S01T1W30P1500 康师傅茉莉蜜茶瓶装500毫升',
# 'MKF339F023S01T1W30P1500 康师傅蜂蜜柚子茶瓶装500毫升',
# 'MKS164F112S01T1W7P1500 康师傅酸梅汤瓶装500毫升',
# 'YBS345F008S01T1W7P1440 怡宝葡萄假日瓶装440毫升',
# 'YBS176F057S03T1W9P11555 怡宝饮用纯净水瓶装1555毫升',
# 'YBS176F057S01T1W9P1550 怡宝饮用纯净水瓶装550毫升',
# 'JTS195F057S03T1W9P1560 景田饮用纯净水瓶装560毫升',
# 'WLJS235F001S01T1W46P1500 王老吉凉茶植物饮料瓶装500毫升',
# 'WLJS235F001S01T1W46P2310 王老吉凉茶植物饮料罐装310毫升',
# 'PCS010F012S01T1W2P11000 百事可乐瓶装1000毫升',
# 'PCS010F012S01T1W2P11250 百事可乐瓶装1250毫升',
# 'PCS010F012S01T1W2P11600 百事可乐瓶装600毫升',
# 'PCS010F012S01T1W2P2330 百事可乐罐装330毫升',
# 'BWS327F202S01T1W49P2550 百威淡色拉格啤酒550毫升',
# 'JTS194F057S03T1W9P1570 百岁山天然矿泉水瓶装570毫升',
# 'RBS012F001S03T1W6P2250 红牛罐装250毫升',
# 'TYS161F036S01T1W15P1500 统一冰红茶柠檬味瓶装500毫升',
# 'TYS252F177S01T1W15P1480 统一小茗同学冷泡茶青柠红茶瓶装480毫升',
# 'TYS254F227S01T1W7P1500 统一海之言海盐柠檬水饮料瓶装500毫升',
# 'TYS251F101S02T1W16P1500 统一绿茶低糖茉莉味瓶装500毫升',
# 'TYS014F001S01T1W4P1500 统一阿萨姆原味奶茶瓶装500毫升',
# 'WTS256F036S01T1W29P1500 维他柠檬茶瓶装500毫升',
# 'WTS256F051S01T1W30P1500 维他柠檬茶蜜桃茶瓶装500毫升',
# 'MNDS015F019S01T1W2P1600 美年达橙味瓶装600毫升',
# 'MZYS264F001S01T1W7P11250 美汁源果粒橙瓶装1250毫升',
# 'FDS023F019S03T1W2P1500 芬达橙味(扭扭瓶)瓶装500毫升',
# 'DLYS278F092S01T1W16P1500 达利园青梅绿茶瓶装500毫升',
# 'DNNS016F030S01T1W6P1600 达能脉动水蜜桃口味瓶装600毫升',
# 'DNNS016F034S01T1W6P1600 达能脉动菠萝口味瓶装600毫升',
# 'DNNS016F206S01T1W6P1600 达能脉动雪柚橘子口味瓶装600毫升',
# 'DNNS016F037S01T1W6P1600 达能脉动青柠口味瓶装600毫升',
# 'XBS017F036S01T1W2P1600 雪碧清爽柠檬味瓶装600毫升',
# 'XBS017F036S01T1W2P2330 雪碧清爽柠檬味罐装330毫升',
# 'XHS304F202S01T1W49P1500 雪花8度勇闯天涯瓶装500毫升',
# 'XHS304F202S01T1W49P2500 雪花8度勇闯天涯罐装500毫升',
# 'QDS355F202S01T1W49P2500 青岛啤酒经典罐装500毫升',]

# replace_names=[
#     'K0ANVCF00010001000 Never Coffee-咖啡饮料-拿铁咖啡-270ml',
#     'K0ANVCF00020001000 Never Coffee-咖啡饮料-燕麦咖啡-270ml',
#     'K0ANVCF00030001000 Never Coffee-咖啡饮料-美式咖啡-270ml',
#     'K1AWXR00010001000 外星人-电解质水-荔枝海盐-电解质饮料-PET瓶-500ml',
#     'K1AWXR00020001000 外星人-电解质水-青柠-电解质饮料-PET瓶-500ml',
#     'K1AWXR00030001000 外星人-电解质水-西柚-电解质饮料-PET瓶-500ml',
#     'K1AWXR00040001000 外星人-能量饮料-经典-能量饮料-PET瓶-350ml',
#     'K1AWXR00040003000 外星人-能量饮料-经典-能量饮料-铝罐-330ml',
#     'K1AWXR00050001000 外星人-能量饮料-马黛茶-能量饮料-PET瓶-350ml',
#     'K1AWXR00050003000 外星人-能量饮料-马黛茶-能量饮料-铝罐-330ml',
#     'K1AWXR00070001000 外星人-能量饮料-乳酸菌-能量饮料-PET瓶-350ml',
#     'K1AWXR00070003000 外星人-能量饮料-乳酸菌-能量饮料-铝罐-330ml',
#     'K1AWXR00080001000 外星人-能量饮料-西柚-能量饮料-PET瓶-350ml',
#     'K1AWXR00080003000 外星人-能量饮料-西柚-能量饮料-铝罐-330ml',
#     'K1AWXR00090001000 外星人-能量饮料-彩虹独角兽-PET瓶-350ml',
#     'K1AYQSL00180001000 元气森林-元气森林-白桃-苏打气泡水-PET瓶-1250ml -福気',
#     'K1AYQSL00180003000 元气森林-元气森林-白桃-苏打气泡水-PET瓶-480ml',
#     'K1AYQSL00180007000 元气森林-元气森林-白桃-苏打气泡水-铝罐-200ml',
#     'K1AYQSL00190001000 元气森林-元气森林-卡曼橘-苏打气泡水-PET瓶-480ml',
#     'K1AYQSL00190005000 元气森林-元气森林-卡曼橘-苏打气泡水-铝罐-200ml',
#     'K1AYQSL00210001000 元气森林-元气森林-荔枝-苏打气泡水-PET瓶-480ml',
#     'K1AYQSL00220001000 元气森林-元气森林-海盐菠萝-苏打气泡水-PET-480ml',
#     'K1AYQSL00230001000 元气森林-元气森林-青瓜-苏打气泡水-PET瓶-480ml',
#     'K1AYQSL00240002000 元气森林-元气森林-乳酸菌-苏打气泡水-PET瓶-480ml',
#     'K1AYQSL00250002000 元气森林-元气森林-酸梅汁-苏打气泡水-PET瓶-480ml',
#     'K1AYQSL00260001000 元气森林-元气森林-樱花白葡萄-苏打气泡水-PET瓶-480ml',
#     'K1AYQSL00290001000 元气森林-燃茶-草莓茉莉-无糖乌龙茶-PET瓶-500ml',
#     'K1AYQSL00300001000 元气森林-燃茶-醇香-无糖乌龙茶-PET瓶-500ml',
#     'K1AYQSL00310001000 元气森林-燃茶-桃香-无糖乌龙茶-PET瓶-500ml',
#     'K1AYQSL00320001000 元气森林-燃茶-玄米-无糖乌龙茶-PET瓶-500ml',
#     'K1AYQSL00340002000 元气森林-元气森林乳茶-咖啡拿铁-乳茶-PET瓶-450ml',
#     'K1AYQSL00350002000 元气森林-元气森林乳茶-茉香奶绿-乳茶-PET瓶-450ml',
#     'K1AYQSL00360002000 元气森林-元气森林乳茶-浓香原味-乳茶-PET瓶-450ml',
#     'K1AYQSL00410001000 元气森林-元气森林-夏黑葡萄-苏打气泡水-PET瓶-480ml',
#     'K1AYQSL00420001000 元气森林-满分-西柚-果汁微气泡-PET瓶-380ml',
#     'K1AYQSL00430001000 元气森林-满分-夏黑葡萄-果汁微气泡-PET瓶-380ml',
#     'K1AYQSL00440001000 元气森林-满分-王林青苹果-果汁微气泡-PET瓶-380ml',
#     'K1AYQSL00450001000 元气森林-元气森林-王林青苹果-苏打气泡水-PET瓶-480ml',
#     'K1AYQSL00490001000 元气森林-燃茶-焙火-无糖乌龙茶-PET瓶-500ml',
#     'K1AYQSL00670001000 元气森林-元气森林-百香果-苏打气泡水-PET-480ml',
#     'K1AYQSL00680001000 元气森林-元气森林-石榴红树莓-苏打气泡水-PET-480ml',
#     'K1AYQSL00690001000 元气森林-纤茶-玉米须茶-PET-500ml',
#     'K1AYQSL00710001000 元气森林-元气森林-红香酥梨-苏打气泡水-PET-480ml',
#     'bottle',
#     'row',
#     'DPS002F002S01T1W6P1250 东鹏特饮维生素功能饮料瓶装250毫升',
#     'DPS002F002S01T1W6P1500 东鹏特饮维生素功能饮料瓶装500毫升',
#     'DPS002F002S01T1W6P2250 东鹏特饮维生素功能饮料罐装250毫升',
#     'LHS061F002S01T1W6P1380 乐虎氨基酸维生素功能饮料瓶装380毫升',
#     'YL0S358F240S01T1W2P1500 依仑大白梨汽水500ml',
#     'YL0S359F244S01T1W2P1500 依仑大连汽水500ml',
#     'YL0S359F242S01T1W2P1500 依仑大连汽水桔子味500ml',
#     'YL0S359F241S01T1W2P1500 依仑大连汽水苹果味500ml',
#     'YL0S359F021S01T1W2P1500 依仑大连荔枝味碳酸饮料500ml',
#     'YL0S360F008S01T1W2P1500 依仑小蜜蜂葡萄果味型碳酸饮料500ml',
#     'JLBS086F114S01T1W6P1560 健力宝运动饮料橙蜜味瓶装560毫升',
#     'BWSS361F243S03T1W2P9380 八王寺山楂香梨维生素果汁汽水380ml',
#     'BWSS361F021S03T1W2P9380 八王寺荔枝维生素果汁汽水380ml',
#     'BWSS361F248S03T1W2P9380 八王寺香橙维生素果汁汽水380ml',
#     'NFSS025F001S03T1W17P1500 农夫山泉东方树叶茉莉花茶瓶装500毫升',
#     'NFSS029F058S01T1W6P1550 农夫山泉尖叫多肽型瓶装550毫升',
#     'NFSS029F005S01T1W6P1550 农夫山泉尖叫纤维型瓶装550毫升',
#     'NFSS030F036S01T1W18P1445 农夫山泉水溶C100柠檬味复合果汁饮料瓶装445毫升',
#     'NFSS030F240S01T1W7P1445 农夫山泉水溶C100西柚味复合果汁饮料瓶装445毫升',
#     'NFSS032F064S01T1W24P1500 农夫山泉维他命水柑橘风味瓶装500毫升',
#     'NFSS026F049S01T1W16P1500 农夫山泉茶π柚子绿茶瓶装500毫升',
#     'NFSS026F036S01T1W15P1500 农夫山泉茶π柠檬红茶瓶装500毫升',
#     'NFSS026F051S01T1W10P1500 农夫山泉茶π蜜桃乌龙茶瓶装500毫升',
#     'NFSS026F009S01T1W17P1500 农夫山泉茶π西柚茉莉花茶瓶装500毫升',
#     'NFSS024F057S03T1W9P1550 农夫山泉饮用天然水瓶装550毫升',
#     'JDBS096F001S01T1W30P1550 加多宝凉茶植物饮料瓶装550毫升',
#     'JDBS096F001S01T1W30P2310 加多宝凉茶植物饮料罐装310毫升',
#     'CCS003F001S01T1W2P1300 可口可乐瓶装300毫升',
#     'CCS003F001S01T1W2P1600 可口可乐瓶装600毫升',
#     'CCS003F001S01T1W2P2330 可口可乐罐装330毫升',
#     'CCS005F001S01T1W2P1500 可口可乐零度瓶装500毫升',
#     'HQZS108F001S01T1W30P1550 和其正凉茶瓶装550毫升',
#     'DYS362F244S01T1W2P9520 大窑嘉宾果味汽水520ml',
#     'DYS362F244S01T1W2P1520 大窑嘉宾果味汽水PET520ml',
#     'DYS363F064S01T1W2P9520 大窑橙诺520ml',
#     'DYS363F064S01T1W2P1520 大窑橙诺PET520ml',
#     'WHHS342F125S01T1W32P1450 娃哈哈AD钙奶瓶装450毫升',
#     'WHHS329F125S01T1W32P1500 娃哈哈营养快线果汁牛奶瓶装500毫升',
#     'WHHS133F001S03T1W9P1596 娃哈哈饮用纯净水瓶装596毫升',
#     'HBLS364F245S01T1W2P9200 宏宝莱果汁鲜山楂味饮料玻璃瓶200ml',
#     'HBLS364F242S01T1W2P9200 宏宝莱果汁鲜桔子味汽水饮料玻璃瓶200ml',
#     'HBLS364F019S01T1W2P9200 宏宝莱果汁鲜脐橙味饮料玻璃瓶200ml',
#     'HBLS348F246S01T1W7P9300 宏宝莱生榨山水蜜桃果味饮料300ml',
#     'HBLS348F247S01T1W7P9300 宏宝莱生榨蓝莓苹果复合果味饮料300ml',
#     'HBLS348F248S01T1W7P9300 宏宝莱生榨香橙果肉饮料300ml',
#     'HBLS365F249S01T1W2P9550 宏宝莱老汽水550ml',
#     'HBLS366F250S01T1W32P1445 宏宝莱花生露植物蛋白饮料445ml',
#     'HBLS366F250S01T1W32P9200 宏宝莱花生露植物蛋白饮料玻璃瓶200ml',
#     'HBLS367F021S01T1W2P9200 宏宝莱荔枝味汽水饮料玻璃瓶200ml',
#     'HBLS367F021S01T1W2P2330 宏宝莱荔枝汽水330ml铝罐',
#     'MKS138F100S01T1W7P11000 康师傅冰糖雪梨瓶装1000毫升',
#     'MKS138F100S01T1W7P1500 康师傅冰糖雪梨瓶装500毫升',
#     'MKS161F036S01T1W15P11000 康师傅冰红茶柠檬口味瓶装1000毫升',
#     'MKS161F036S01T1W15P1500 康师傅冰红茶柠檬口味瓶装500毫升',
#     'MKS166F001S03T1W9P1550 康师傅喝开水熟水饮用水塑料瓶550毫升',
#     'MKS164F001S02T1W16P1500 康师傅绿茶低糖瓶装500毫升',
#     'MKS164F001S01T1W16P11000 康师傅绿茶瓶装1000毫升',
#     'MKS164F147S01T1W30P11000 康师傅茉莉清茶瓶装1000毫升',
#     'MKS164F147S01T1W30P1500 康师傅茉莉清茶瓶装500毫升',
#     'MKS164F148S01T1W30P11000 康师傅茉莉蜜茶瓶装1000毫升',
#     'MKS164F148S01T1W30P1500 康师傅茉莉蜜茶瓶装500毫升',
#     'MKF339F023S01T1W30P1500 康师傅蜂蜜柚子茶瓶装500毫升',
#     'MKS164F112S01T1W7P1500 康师傅酸梅汤瓶装500毫升',
#     'YBS345F008S01T1W7P1440 怡宝葡萄假日瓶装440毫升',
#     'YBS176F057S03T1W9P11555 怡宝饮用纯净水瓶装1555毫升',
#     'YBS176F057S01T1W9P1550 怡宝饮用纯净水瓶装550毫升',
#     'WZS371F240S01T1W2P9310 旺仔大白梨果味型碳酸饮料310ml',
#     'JTS195F057S03T1W9P1560 景田饮用纯净水瓶装560毫升',
#     'WLJS235F001S01T1W46P1500 王老吉凉茶植物饮料瓶装500毫升',
#     'WLJS235F001S01T1W46P2310 王老吉凉茶植物饮料罐装310毫升',
#     'ZZ0S368F251S01T1W2P2330 珍珍桃味汽水330ml',
#     'ZZ0S368F242S01T1W2P2330 珍珍桔味汽水330ml',
#     'ZZ0S368F021S01T1W2P2330 珍珍荔枝味汽水330ml',
#     'PCS010F012S01T1W2P11000 百事可乐瓶装1000毫升',
#     'PCS010F012S01T1W2P11250 百事可乐瓶装1250毫升',
#     'PCS010F012S01T1W2P11600 百事可乐瓶装600毫升',
#     'PCS010F012S01T1W2P2330 百事可乐罐装330毫升',
#     'BWS327F202S01T1W49P2550 百威淡色拉格啤酒550毫升',
#     'JTS194F057S03T1W9P1570 百岁山天然矿泉水瓶装570毫升',
#     'QYS369F252S01T1W2P1350 秋林格瓦斯PE瓶350ml',
#     'QYS370F253S01T1W2P1350 秋林黑格瓦斯PE瓶350ml',
#     'RBS012F001S03T1W6P2250 红牛罐装250毫升',
#     'TYS161F036S01T1W15P1500 统一冰红茶柠檬味瓶装500毫升',
#     'TYS252F177S01T1W15P1480 统一小茗同学冷泡茶青柠红茶瓶装480毫升',
#     'TYS254F227S01T1W7P1500 统一海之言海盐柠檬水饮料瓶装500毫升',
#     'TYS251F101S02T1W16P1500 统一绿茶低糖茉莉味瓶装500毫升',
#     'TYS014F001S01T1W4P1500 统一阿萨姆原味奶茶瓶装500毫升',
#     'WTS256F036S01T1W29P1500 维他柠檬茶瓶装500毫升',
#     'WTS256F051S01T1W30P1500 维他柠檬茶蜜桃茶瓶装500毫升',
#     'MNDS015F019S01T1W2P1600 美年达橙味瓶装600毫升',
#     'MZYS264F001S01T1W7P11250 美汁源果粒橙瓶装1250毫升',
#     'FDS023F019S03T1W2P1500 芬达橙味(扭扭瓶)瓶装500毫升',
#     'DLYS278F092S01T1W16P1500 达利园青梅绿茶瓶装500毫升',
#     'DNNS016F030S01T1W6P1600 达能脉动水蜜桃口味瓶装600毫升',
#     'DNNS016F034S01T1W6P1600 达能脉动菠萝口味瓶装600毫升',
#     'DNNS016F206S01T1W6P1600 达能脉动雪柚橘子口味瓶装600毫升',
#     'DNNS016F037S01T1W6P1600 达能脉动青柠口味瓶装600毫升',
#     'XBS017F036S01T1W2P1600 雪碧清爽柠檬味瓶装600毫升',
#     'XBS017F036S01T1W2P2330 雪碧清爽柠檬味罐装330毫升',
#     'XHS304F202S01T1W49P1500 雪花8度勇闯天涯瓶装500毫升',
#     'XHS304F202S01T1W49P2500 雪花8度勇闯天涯罐装500毫升',
#     'QDS355F202S01T1W49P2500 青岛啤酒经典罐装500毫升',
# ]

for m_name in model.names:
    if len(m_name.split(' ')) > 1:  # 重编码前的
        sp = "%s " % m_name.split(' ')[0]
        name = m_name.split(sp)[1]

        if name == "东鹏大咖®咖啡饮料":  # 这段后续应该用不到了
            if "330" in m_name:
                name = "%s%s毫升" % (name, "330")
            elif "500" in m_name:
                name = "%s%s毫升" % (name, "500")

    else:
        name = m_name

    if name in ['bottle', 'row']:
        name = m_name
        replace_names.append(name)
    else:
        replace_names.append("%s %s" % (name_to_code_dict[name], name))

model.names = replace_names
torch.save({'model': model}, output_model_path)  # 把修改后的模型保存
strip_optimizer(output_model_path)  # 精简模型，这会覆盖原模型

print(output_model_path)
